state machine in python